import csv

with open('data/predict_data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(["id", "pest_name", "next_day", "next_two_day", "next_three_day"])
    writer.writerow([1, "小灰蝶", "20.90236817", "21.21619192", "21.66220647"])
    writer.writerow([2, "荔枝椿", "23.15075813", "23.49235306", "23.521714"])
    writer.writerow([3, "金龟子", "15.23099831", "15.18319978", "15.13836018"])
    writer.writerow([4, "龟背天牛", "29.65156701", "29.59558583", "29.54594635"])